# 1.准备数据  

```
python preprocess_RGBTCC.py
python preprocess_DroneRGBT.py

rm -rf data/DroneRGBT/val/
cp data/DroneRGBT/test data/DroneRGBT/val -r
```

# 2.训练  

```
python train.py \
    --data-dir ./data/RGBTCC \
    --save-dir ./pth_RGBTCC \
    --lr 1e-5 \
    --val-start 20 \
    --device 0

python train.py \
    --data-dir ./data/DroneRGBT \
    --save-dir ./pth_DroneRGBT \
    --lr 1e-5 \
    --val-start 20 \
    --device 0

```

# 3.测试

```
python test.py \
    --data-dir ./data/RGBTCC \
    --save-dir ./pth_RGBTCC/0912-050201 \
    --model "best_model_0.pth" \
    --device 0


python test.py \
    --data-dir ./data/DroneRGBT \
    --save-dir ./pth_DroneRGBT/0912-050508 \
    --model "best_model_0.pth" \
    --device 0
```